package club.xiaojiawei.hash;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 2022/4/25 下午10:13
 * @question 49.字母异位词分组
 * @description 给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
 * 字母异位词 是由重新排列源单词的字母得到的一个新单词，所有源单词中的字母通常恰好只用一次。
 */
public class AlphabeticWordGrouping49 {

    public static void main(String[] args) {
        String[] strings = {"eat", "tea", "tan", "ate", "nat", "bat"};
        List<List<String>> result = groupAnagrams(strings);
        result.forEach(strings1 -> {
            strings1.forEach(System.out::println);
            System.out.println("++++++++");
        });
    }

    /**
     * HashMap+字符串排序
     * @param strs
     * @return
     */
    public static List<List<String>> groupAnagrams(String[] strs) {
        HashMap<String, ArrayList<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String s = new String(chars);
            ArrayList<String> list = map.getOrDefault(s, new ArrayList<String>());
            list.add(str);
            map.put(s, list);
        }
        return new ArrayList<>(map.values());
    }

    /**
     * 民间-计数+排序
     * @param strs
     * @return
     */
    public List<List<String>> groupAnagrams2(String[] strs) {
        HashMap<String, List<String>> anagramsGroup = new HashMap<>();
        List<String> list;
        for(String s : strs){
            String code = encode(s);
//            anagramsGroup.putIfAbsent(code, new ArrayList<>());
//            anagramsGroup.get(code).add(s);
            if ((list = anagramsGroup.get(code)) == null){
                anagramsGroup.put(code, new ArrayList<>(){{add(s);}});
            }else {
                list.add(s);
            }
        }
        return new ArrayList<>(anagramsGroup.values());
    }

    /**
     * 利用每个字符的出现次数进行编码
     * @param s
     * @return
     */
    String encode(String s){
        char[] code = new char[26];
        for(char c : s.toCharArray()){
            int delta = c - 'a';
            code[delta]++;
        }
        return new String(code);
    }

}
